CoBa: Cost-Effective Test-Time Scaling via Compute-Balanced Routing
CoBa:通过计算均衡路由实现具成本效益的测试时扩展
🧭 LLM 测试时计算优化
School of Mathematics and Statistics, Changsha University of Science and Technology, Changsha, China · School of Biomedical Engineering, Shanghai Jiao Tong University, Shanghai, China
提出计算均衡路由策略,动态分配生成与验证预算,以更少计算匹配多数投票准确率。
📌 工作参考价值(4 点)
- 多候选验证的预算分配:在生成式推荐或搜索中,当需要从多个候选 query/item 中投票选择时,可先使用廉价验证器(如轻量级打分模型)过滤大量候选,仅将高不确定性或高价值候选送入昂贵的大模型验证,大幅减少 LLM 调用成本。
- Agent 循环中的动态路由:在涉及规划-执行-评估的 Agent 系统中,可借鉴 CoBa 的“生成 vs 验证 vs 停止”决策框架,由路由模块实时判断下一步应生成更多方案、调用评估器,还是直接输出结果,避免无效计算。
- 在线服务成本控制:对于对延迟敏感的电商实时推荐,可基于 CoBa 的停止规则设定置信度阈值,当廉价验证结果一致性高时提前终止,节省的 token 可分配给更复杂的用户意图理解场景。
- 粗-精验证管道设计:构建类似于 CoBa 的两级验证结构,第一级用规则或小模型快速打分,第二级用大模型深挖,这在商品标题改写、用户评论情感分析等场景中可直接复用,提升吞吐量。